Visualization using Python code